﻿// three_nums_sum.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
// 三数之和

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<vector<int>> threeSum(vector<int>& nums) {
    vector<vector<int>> result;
    sort(nums.begin(), nums.end());
    if (nums[0] > 0) return result;

    int nums_size = nums.size();
    
    for (int i = 0; i < nums.size(); i++)
    {
        if (i>=1 && nums[i] == nums[i - 1]) continue;
        int left = i + 1;
        int right = nums_size - 1;

        while (left < right)
        {
            if (nums[i] + nums[left] + nums[right] < 0)
            {
                left++;
            }
            else if (nums[i] + nums[left] + nums[right] > 0)
            {
                right--;
            }
            else
            {
                vector<int>result_item{ nums[i], nums[left], nums[right] };
                result.push_back(result_item);
                while (left < right && nums[right] == nums[right - 1])
                {
                    right--;
                }
                while (left < right && nums[left] == nums[left + 1])
                {
                    left++;
                }
                left++;
                right--;
            }
        }
    }
    return result;
}

int main()
{
    vector<int> nums{ -1,0,1,2,-1,-4 };
    //vector<int> nums{ 0, 0, 0 };
    vector<vector<int>> result;
    result = threeSum(nums);
    for (auto& outer : result)
    {
        for (auto& inner : outer)
        {
            cout << inner << ' ';
        }
        cout << '\n';
    }
}

// 运行程序: Ctrl + F5 或调试 >“开始执行(不调试)”菜单
// 调试程序: F5 或调试 >“开始调试”菜单

// 入门使用技巧: 
//   1. 使用解决方案资源管理器窗口添加/管理文件
//   2. 使用团队资源管理器窗口连接到源代码管理
//   3. 使用输出窗口查看生成输出和其他消息
//   4. 使用错误列表窗口查看错误
//   5. 转到“项目”>“添加新项”以创建新的代码文件，或转到“项目”>“添加现有项”以将现有代码文件添加到项目
//   6. 将来，若要再次打开此项目，请转到“文件”>“打开”>“项目”并选择 .sln 文件
